numpy==1.19.5
scipy==1.5.4
torch==1.8.0+cu111
torchvision==0.9.0+cu111
tqdm==4.64.1